# 导包

from tensorflow.keras.models import load_model
from utils.data_generator import test_generator

import os
os.environ['CUDA_VISIBLE_DEVICES'] = '1'

#读取测试集

test_gen = test_generator(
    data_dir='../dataset/test',
    target_size=(64, 64),
    batch_size=32,
    class_mode='categorical'
)


"""
模型载入 tf.keras.models.load_model
参数：
-filepath

模型验证tf.keras.Sequential.evaluate
参数:
-x 


"""

# 模型保存的相对路径

model_path = '../models/model-2020-10-28-21-52-12'

# 载入模型

loaded_model = load_model(filepath=model_path)

# 模型验证测试集

loss, accuracy = loaded_model.evaluate(x=test_gen)



